import { ApiProperty } from '@nestjs/swagger'

import { DeviceEntity } from './device.entity'

export class DeviceInfo extends DeviceEntity {
  @ApiProperty({ type: [Number] })
  menuIds: number[]
}

export class DeviceToken {
  constructor(uid: string, token: string) {
    this.token = token
    this.uid = uid
  }

  @ApiProperty({ description: 'Token' })
  token: string

  @ApiProperty({ description: '设备ID' })
  uid: string
}
